python function with local variable